#!/bin/bash
export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libhwloc.so
export ENABLE_INIT_ON_CPU=1
export XLA_FLAGS="--xla_backend_extra_options='xla_dtu_mixed_precision'"
export TF_XLA_FLAGS="--tf_xla_auto_jit=-1 --tf_xla_min_cluster_size=4"
export DTU_UMD_FLAGS="ib_pool_size=41943040"
export ENABLE_RDMA=true
export ECCL_TOPOLOGY=2
export ENFLAME_AUTO4C=true
